#ifndef ASCEND_INDEX_H
#define ASCEND_INDEX_H

extern "C" {
    void init(int device_id, int dim);

    void insert_gallery(int baseSize, float* feat);

    void search(int n, float* queryFeat, int topK, float* scores, int64_t *idxs);

    int getBaseSize();
}

#endif